package controller;

import entidade.Treino;
import java.util.List;
import javax.faces.model.ListDataModel;
import org.primefaces.model.SelectableDataModel;

/**
 *
 * @author Thais
 */
public class ListaTreinoDataModel extends ListDataModel<Treino> implements SelectableDataModel<Treino> {

    public ListaTreinoDataModel(List<Treino> data) {
        super(data);
    }

    @Override
    public Treino getRowData(String rowKey) {
        List<Treino> treinos = (List<Treino>) getWrappedData();

        for (Treino treino : treinos) {
            if (treino.getIdTreino().equals(Long.parseLong(rowKey))) {
                return treino;
            }
        }
        return null;
    }

    @Override
    public Object getRowKey(Treino t) {
        System.out.println("get row key");
        return t.getIdTreino();
    }
}
